package de.tum.in.protpred;

import de.tum.in.protpred.classification.TMHClassifier;
import de.tum.in.protpred.eval.TMHEvaluater;

public class Main {
	public static void main(String[] args) {
		String[] options = {"-S 0", "-K 2", "-D 3", "-G 0.0", "-R 0.0", "-N 0.5", "-M 40.0", "-C 1.0", "-E 0.001", "-P 0.1"};
		String trainingSet = "/home/jonas/Desktop/features_manual_0.arff";
		String testSet = "/home/jonas/Desktop/features_manual_0.arff";
		try {
			TMHClassifier blub = new TMHClassifier(trainingSet, testSet, options);
			System.out.println("Building classifier");
			blub.buildClassifier(trainingSet);
//			System.out.println("Loading old model");
//			blub.loadModel("/home/jonas/Desktop/model.svm");
			
			String outputPath = "/home/jonas/Desktop/out.txt";
			System.out.println("Writing output");
			blub.createOutput(outputPath);
			//			System.out.println("Distribution");
			//			blub.getDistribution();
			
//			System.out.println("Saving trained model");
//			blub.saveModel("/home/jonas/Desktop/model.svm");
			
			
			System.out.println("Evaluation scores");
			TMHEvaluater.calcPerResidueScores(blub.getConfusionMatrix(), blub.getEvaulation() ,"/home/jonas/Desktop/out.scores");
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
}
